#!/bin/bash

config_path=conf/default.yaml
MODEL=pwgan_thorsten
train_output_path=/home/aistudio/exps/$MODEL/train

FLAGS_cudnn_exhaustive_search=true \
FLAGS_conv_workspace_size_limit=4000 \
DUMP_DIR=/home/aistudio/exps/$MODEL
python3 ${BIN_DIR}/train.py \
    --train-metadata=${DUMP_DIR}/train/norm/metadata.jsonl \
    --dev-metadata=${DUMP_DIR}/dev/norm/metadata.jsonl \
    --config=${config_path} \
    --output-dir=${train_output_path} \
    --ngpu=1
